from django.views import View
from django.http import JsonResponse
from django.core.cache import cache
from .models import Area
import logging

logger = logging.getLogger('django')


class AreasView(View):
    """省市区数据"""
    def get(self, request):
        """提供省市区数据"""

        # 提供省份数据
        try:
            # 先查询缓存数据
            province_list = cache.get('province_list')
            # 如果没有缓存数据，则查询数据库数据
            if not province_list:
                # 查询省份数据
                province_model_list = Area.objects.filter(parent__isnull=True)
                # 序列化省级数据
                province_list = []
                for province_model in province_model_list:
                    province_list.append({'name': province_model.name, 'id': province_model.id})
        except Exception as e:
            logger.error(e)
            return JsonResponse({'code': 400, 'errmsg': '省份数据错误'})

        # 保存缓存数据
        cache.set('province_list', province_list, 24 * 3600)

        # 响应省份数据
        return JsonResponse({'code': 200, 'errmsg': '', 'province_list': province_list})


class SubAreasView(View):
    """省市区数据"""

    def get(self, request, pk):
        """提供省市区数据"""

        # 提供市或区数据
        try:
            # 先查询缓存数据
            sub_data = cache.get('sub_area_' + pk)
            # 如果没有缓存数据，则查询数据库数据
            if not sub_data:
                parent_model = Area.objects.get(pk=pk)  # 查询市或区的父级
                sub_model_list = parent_model.subs.all()

                # 序列化市或区数据
                sub_list = []
                for sub_model in sub_model_list:
                    sub_list.append({'name': sub_model.name, 'id': sub_model.id})

                sub_data = {
                    "id": parent_model.id,  # 父级pk
                    "name": parent_model.name,  # 父级name
                    "subs": sub_list  # 父级的子集
                }
        except Exception as e:
            logger.error(e)
            return JsonResponse({'code': 400, 'errmsg': '市区数据错误'})

        # 保存缓存数据
        cache.set('sub_area_' + pk, sub_data, 3600)
        # 响应市或区数据
        return JsonResponse({'code': 200, 'errmsg': '', 'sub_data': sub_data})
